Correct unit tests. And update the README