Added caffe classify logic