Add data_partition variable