Updated code

Updated README.md
Initial code for dataset