added problem 1 solution