code refactoring
Update Readme