Add more comment  to code and README.md