The development of a cognitive rehabilitation task for mice