Minor tidyup