added correct class