works
(have to change paths to train and test corpus)