Added another Main for single image testing