Update solution.rb