Changed the number of splits and removed the debugging print statement.
Removed the FeaturePartitioner (which should have never been in the repo) and added the ByBucketPartitioner (which currently has a bug).