Update three-way-partition.rb