Assignment 1 update :)