Add new python demo code