Updates for SARSA implementation