Classes of units now inherit from Unit instead of just having a different string for their class