Improve the documentation in the README