fixing python script