Updated README, bug fix