Cleaned up Partitioning example