Update README, add unit tests