revise readme and part of code