updated Python code
input raster