adding missing class