[+] Refactoring
[+] Add bayes classifier on the top