Really added bayes algorithm, started work with testing