Updated the example code