Minor changes to matos_hw5.rb