Update README.md and fixed a small bug