Add test for add/remove partition?