Add some type hints.
Merge pull request #211 from banjiewen/master

Add some type hints