fixed small bugs in example