Add documentation for stage_elem