added missing class