Typo in README.
Minor Changes to logic