Update outdated code sample