Adding forgotten component class