Added the missing class