small change on the example code