Bug fixed. Add test case
Modify readme