chore: remove unecessary imports and add comment in _Driver