add Problem Set 2 problem 1 solution