Fix readme and unit test