Data science course's homework