add test code & fix its bug
0.1.1