Added signatures for classification stage