problem 1 in golang