Refactoring the code to fit the Python-CI pipeline

The Black check and version check break and stop the pipeline prematurely.