Update switch.py

fixed formatting.