Add Protocol class