Yeah this section is annoying. I still need to call updateDisplay though.