made some changes to new code

changed import to fix weird bug in eclipse (where it was not
recognizing nltk.tokenize.punkt

made some optimizations to code of collocations.py