q learning agent works also for red, blue