Change it completely to python