update README
fix the main bug in checking the convergence and some better
reorganization of code;