Changed to if statement