Added 4th solution to step2.rb