Minor code adjustments in the example code