Improve the docs
Cleanup formatting of docs