Instead of summing the cost, we should show partition-wise cost in order to check the load balancing. Same for communication.