Unit class added