Change to using typed docstrings.

Thanks.