Updated Sample Code

Minor Tweaks.