add class to protocol pkg