More auto_partition.py