Add type hints (why not?)