added adding trains. Final commit before refactoring