Tweak example code a little bit