Added second if statement

Added second if statement