Managing display logic with classes