Small refactor.
Update README.