Update README
Refine the code.