add if-and-else