Fix Puzzle Update Function