It would be cool if the puzzle mechanic continued over into the second level --
It should have. Did you not encounter any puzzles in the second level?
