Added missing class