Final tidy up