Update simple_switch.py