Updating code sample