add IF statement