Lemmatization working a little better