fixed code to add disconnect in some parts