Minor tidy up 2