Working on python example