Updated the code example