adding Naive Baye's class and adding methods to Main.java