Add test files as well