Final code tidy up