Fix problems with OOP on Python