Data science Coursera Assignment