Small tweaks and documentation