Update episode.py