Update corpus.py