update README && optimize code