Fixed unit test
Added random forest