Update puzzle.rb