Add missing source files