Small editions to example code