Updated README code sample
Updated README