Added protocol selection

Thanks Shea!