this is student task from data scientist course at Coursera