Fixed the comparison; added second option

- Fixed bugs with the comparison with each document in the dat set that
  we provide.
- Added a proper second option to provide your own data set and compare
  it with a document for comparison.