python code for dataScience course