My homework for GA Data Science class