Create balanced_partition.rb