Data Science coursea exercise