Better code generation for switch.