adding if-then-else