Improved demo code

...

Signed-off-by: Andrew D. Horchler <adh9@case.edu>