Updated if section