Updated Sample Code

Updated Sample code