Add partition creation steps