Update feature_vector.py