Generalize the code. Fix a Bug.
Modify README.md.