homework for Data Science course through General Assembly