Implemented the image classifier