AI Learns Platformer - little upgrade