Fixed bug where words with zero occurances are added to training set