Adding Image classification script