Change example program accordingly