Add in correct css highlighting when returning to page