solution for problem 2 in go