Made changes to example code.