refine example code
refine example code