Fixed major bug and added prediction capability.
Added unit test to cross validate against iris data set