Unit_Class Update 2

some minor changes.