Improved the keyword extractor