Misc. updates for tidy up