fixing pythong script