Updated Assigment 1 slightly