Add tests for section 6