Trying to progress in the python script