Update puzzle/puzzle.rb