Add README.md and some small improvements