changes to match Kaggle