WEEK 4 ASSIGNMENT SOLUTION OF PYTHON FOR DATA SCIENCE