Improve readme
Increase test coverage