Revert "removing unncessary class keyword in protocol"

This reverts commit 79d5ae9c84625a08f7e44ae43a491c5638400462.