Adding forgotten class