Completed code updates from Lecture 2 slides, ready to start Assignment 1