Assignment #4, Changed assignment_4.rb to solution.rb with different code