Add else if test