Some lab 2 notebook progress, and same code in .py file.