update after hw4, part 1