Update unit.cs

Log class usage