adding the missing class