Updated python example script