Tweak code sample