add classes for socket