The agent is learning, but i think i may have implemented update incorrectly