Fixed the feature vector