Tidy up part 2