Added argument to caffe extract