Added protocol to code.