Homework for data scientist