Add if statements