Added code to generate test files