Improve the README and add some documentation