Update switch.py