improve to support partitioning new disks