Update custom/classes/sort.rb