Fixed take partitions so that you dont assign owner if its already a successor
Cleaned up return partitions