Python code for John Hopkins Regression course