Update README
Improve tests