Updated python sample