Code for the Kaggle competition