Update corpus.py for new dir structure