Add tests and fix issue when splitting into sentences, to grab the minimum number between total sentences and sample size, rather than total original documents and sample size

Add CLI and refactor

Support visualizing both sentences and whole documents. Smooth down color assignment in visualization.