add the missing classes