Code snippet for Python