Update hw4.go