Modify example code