add some comment in initial_partitioning.py