Modified Code and Updated Readme