Enhance type-hints #16
Description
The current codebase uses Dict[str, Any] for many class attributes, which can make the code harder to understand and use. We need to enhance the type hints by replacing Dict[str, Any] with more specific types wherever possible (e.g Dict[str, Parameter])