Solution to w1c3.rb