Add diagram_free