NB, LR prediction added

Added predictions method for NB and Linear regression. Also there is
improved data model.