add if and else