added Unit class