Fix layout of the documentation

Add snippet snippets
Proper code blocks