Add custom code and a conditional