push playing around with kaggle competition, code adapted from tutorial