Update problem2.go