Very slight modification to corpus_reader.py.