Update Code Sample

Update Code Sample