Update wagon_sort.rb