Added revert functionality and test case.
A little cleanup work.
Updated README.md