Fix error in predictions_class variable for dataset 2
Added option for loading the 3rd data set
Added bayes equation