The code for my Kaggle project