refactored updating display a little bit, more centralized