Added the partitioning functionality. Starting to implement data collection for it.