Add support for protocol