Make demo code python syntax