Modification to example program