Updated code and Readme

Updated code and Readme