Solved problem 4 - Quite happy with answer