Add if-else