Refactor for_body func