Further update to code example