setup for classification seperate