assignment 3 python for data science