Added syntax highlighting for Python code block