Started working on problem 3