Finished problem number 4, now adding problem number 5