for Data Science Coursera assignment