updated corpus.py