Improve graph formatting. Add support for larger corpus.