Update learning-switch.py