Added [Protocol] attribute to required classes