Add configuration for partition handling