Revised code example