Small tweaks and docs