Improve script for checking tests to get additional test files