Update sample code per Patrick's suggestion