Added labeled dataset option