added the missing classes