add if/else condition