Update problem1.go