small edits to random forrest