Modified word_feature.py