Update gen_vw_features.py