Started working on problem 2