Rewrote prediction code to use naive bayes' instead of multiple linear regression.