Add partition columns