further tweak example code