This is part of a data science course project for week 3