Updated proto files for full functionality