Improved on the example program...