adding countvec for feature creation and testing on imma dataset