Added missing class file