Add configuration logic