Update README

made some changes to the code & just made updates to reflect that.