Use if/else block