Update

Fix a bug.
Update

Add test cases.