Cool concept! Few ideas:

1. when you hover over puzzles, they could shuffle/rotate a bit, so that there's no easy option to just click puzzles and rotate them appropriately upfront

2. an option to select a group of puzzles and move them, just as you categorize and group puzzles in real life