File for data science assignment