More Documentation and type hint improvements