improve socket

improve socket