improve readme
minor refactoring