Added initial impl of partitioning component