Add the socket class