Update documentation for better rendering to book