imlement function initial partitioning in initial_partitioning.py