Update code sample