Add in else if statement