Update basic_walkthrough.py