Update README, add tests