Added if statemnt