Update code samples to match the source