Add protocol testing