Add needed type annotation. Make partial function not full function.
Merge pull request #35 from andrewbts/patch-1

Add needed type annotation. Make partial function not full function.