Added missing client disconnect logic