Added classification training.
Merge pull request #1 from anoras/classification

Add classification training.