Updated the display so it actually works