working on Kaggle assignment