- Update doctests
- Fix tokenization bug
- Update README