Fixing code sample
minor tweaks