adding file for data science homework