added trainer and clasifier