Remove socket cleanup code