Added the else if statement