Add remove partition support