adding another "if" statement to your code