More tidyup