added in else if statement