Case instead of if/else