Can measures of cognitive function predict locomotor behaviour in complex environments following a traumatic brain injury?