This is for assignment #1 in the data scientist course.