Update README.md

Add more description about the functions of the code.