problem 1 in Go