Refactor code and fix imports

[tabedzki]

Refactored the code to improve readability and removed unnecessary variables and expressions. Fixed incorrect variable references and organized imports and adjusted white space. This pull request aims to improve the overall code quality and maintainability.

Moving forward, we can use ruff to both lint and format our code.

@AlvaroS, can you give your thoughts on this

[Alvalunasan]

Hi Christian, I think this is a great addition.
Do you have a plan on how we could test all changes introduced given the current state of the code?

[tabedzki]

I think the safest thing would be to setup a bunch of tests related to this. However, I think that would require a lot of work to verify that things like this work. There are options to make the formatter do only safe changes to the code, therefore it wouldn't do something like change

def f(a = []): pass # unsafe because every call to `f` would refer to the same list

into

def f(a=None):
    if a is None:
        a = [] # safe because the list gets assigned when the function is called and not when the function is defined
    pass

There is more that I would like to test out before making the plunge on this but having a formatter/linter seems like it would be useful.