Added the image recognition functionality