Modify sockets mapping and add output colors