How to wrap address field output in fieldset?

Hello,

I was wondering how I could go about wrapping a fieldset around the output of addresses for a create/edit form?

I found that if I edit the function _addresses_addressesfieldapi_form in addresses.settings.inc and add an array like so

and then add all the elements of address to the array that it would work, but I would rather not touch the modules files and instead override it from my template.php, I just don't know how..an example would be great.

Thanks