Oh, I need to update display method as well!