Fixed a code error and updated the readme