small tweaks in example code