Wrote frame work for problem 4