Add in protobuf code