Fix rest of code to return in files array.