Add the missing classes