Updated Python Example