Alterations in cognitive flexibility in a rat model of post-traumatic stress disorder