Refactor codebase with Pylint & Black

[JanBenisek]

Before release, refactor the code so it has the same formatting using Black (https://github.com/psf/black) Afterwards, we can add Black checks part of the CI pipeline and enforce code quality