Update display and fix protocol issue