added the missing class