Update puzzle.js and main.js