Tidy up
Problem 4