Bellman Prediction Error for State Value Updates - Coding problem | OneNoughtOne