Added code and updated Readme