Added missing Class