Added context managers in UnitSystem