Updates with my hw4 part 2