Add protocol stuff