Updated if clause