Updated the SARSA agent