Add if and else conditional statement