SQUARE WITH 85 IQ :

ML AGENT THAT OBSERVES THE COLOR