add error check for if-else statement