Finished up naive bayes for this simple example. Added tests as well