Add info in README.md and delete some code for optimization