Updated Sample Code

minor tweaks