Fix "Play Again" button, and simplifying some puzzle loading