created as part of course project for data science certification