added add train helper function