Add IF and pass test