Code style check

[amrit110]

This issue has 2 parts:

• The codebase needs a review to see if it adheres to google code style guide (https://google.github.io/styleguide/pyguide.html).

• For docstrings, the parameter types are documented using type hints, and adding them to the docstring is redundant. These can be removed to keep docstrings clean and reduce redundancy.