First data science assignment git hub repo