Refactored ElectricityDisplay to UnitDisplay