New function for updating the policy added

I have added a new function for finding the optimal policy given
discount factor and the value grid (that's the second last row of figure
4.3)
I think it might be also used for value_iteration!! :)
I have left the previous policy_improvement function commented in case
you want to go back to it or something....