new final project, adjusted kaggle script