Cleaning up puzzle.js