Added and solved problem 4