Updates assignment1, Excercise 2