demo code for python