Add cleanup hook for integration tests