Add type hint
Merge pull request #2 from bakura10/patch-2

Add type hint