Updated code samples to use python syntax highlighting