For data science assignment