Changed else if statement a bit.