Updated Python example