Add source files for test