Q learning (assumimg, need better testing) running correctly