Brain games help restore function after mild TBIs, study finds