Example code for python