Q-Learning kind of works :)