README update and code optimization