Tiny kaggle score improvement
Better kaggle score