Code for Kaggle competition