add code-block to python sample code

Applied proper syntax highlighting to a code sample in the docs