Seperated problem 4 and 5