Same here. Puzzling part after the first few levels is more about figuring out the rules of the game than solving actual puzzles.
This is not bad in general, but for some reason here it felt a bit frustrating.
If it will be scaled into a bigger game in future, I'd suggest adding few more levels for any mechanic before going to the next one.