new partition assignment code working