added nltk code