add main for classification