I think this is a really cool concept. The main point of improvement I'd mention is tightening up the level design, with smaller puzzles and less space in between. Aside from that, just adding a frictionless physics material to the player's collider to avoid getting stuck on walls, and rather than showing only one level on the big screen, then three on the side, maybe give all three colors the same screen real estate.