Improve Documentation

Improve the styling of code examples in the documentation.