Refer to #CODE BLOCK 1# on the Jupyter Notebook