Sample python code