Code sample updated