Can this code work? There are no train or test files either. I haven't figured out exactly what they are for!