Add missing classes