Updated the missing class