Updated example code a little.