- Implement more unit tests
- Consider cleaning up the type-specific parsing
- Add type annotations