Add abort functionality