Update puzzle.lua