Add random forest related code