problem 1 in go lang