add the IF part