added some more code from lecture 3 walkthrough