Add support for socket types