add unit test
edit readme