Fixed find and replace mistakes from last commit.
Fixed a bug, that would cause counting to crash if a training corpus was given as input arg.