solved problem 4 and set up 5