I like that "New puzzle" idea, it'd add some variety. Besides, as the levels go up you're supposed to be better at that thing. Might as well represent that in-game.
