cleaned up protocol graph demo