optimize the code
update README document