functionality is restored. No longer hard-coded for 2 specific files