Finished adding code from Lecture 3 slides. Ready to do Assignment 1.