Docs and small tweaks