Finished training logic for the non-binary case
Added support for the binary counting case
Added code to do classification once classifier is trained.
Finished, ready to submit