Small fixes,
few more train files added