Add type annotations to functions, also the comments for a description of a function should be in form of a docstring.