Fix bug in balance_partition fuction

Reduce NCP again. Results are still larger than paper.