- update readme
- small refacture