updated to new kaggle data