Add missing classes to compile