change in README.md and made the code in the main calls a bit more
pretty