small refactor. readme improvement