Better typing support

The task is to update typing and create custom types. Also, use some type-checker.

types.py:

# Types
SerializedOutput = Union[list, dict]