Finished featurize_data.py