Added if statement