added a main function to test on the dev directory
adding naivebayes trains to the __train__ method