Updade code snippet