Homework task for coursera datascience course