Pre-lab updates of week6.ipynb