Trivial optimization in example code.
Merge pull request #4 from ADmad/master

Trivial optimization in example code.