Code for Kaggle competition March Machine Learning Madness