Add if-then-else statement