Minor changes in the example code