Improve docs

Do a quick once over to spruce up the documentation a little bit.