Added type annotation to some functions to make the code more readable.

Added type annotation to some functions to make the code more readable. #110