Add Unit class