Python Script Using Histogram Equalization Problem