Minor formatting and improved docs