Updated README code part accordingly