Cute little start! Seems like the logic could need a little work. In general, for puzzle games like this, I'd try to decouple the logical game state from visuals, which I think might be the reason behind some of the bugs? Anyway, keep going!
