Short script written with arcpy to be used in an ArcGIS toolbox