Update to the example program

No functional changes. Just changed an import and added a few comments.