Add forest function. Need to implement