Small improvements
Updated README.md