Minor changes to example code