Renamed get_puzzle
Switching between puzzles kinda works now