Sample Python code