Updating Assignment 1. Thought I had already done this...