Python for Data Science week 2 solutions