Misc tidyup