Update problem3.go