adding initial Unit class