Add missing class file