Update unit tests for add_to_section