Update more docs and type hints to use IDatabase