Minor improvements in example