Add cleanup wrapper