Updated if statement