Added IF statement