Add missing generated classes