modify example code

modify example code