Added if block