Update switch_key.py

Minor changes.