Add missing class