Add client.disconnect()