Adding Train directory and data files