fixed the train mode for the Sarsa agent