Adding support for random forests