Add problem 1 solution