Added if tests