Working on socket wrapper