Update readme and Unit test