Add problem 7 in Go