This repo is for data science assignment