part of data science course project