Train potential

@traciegiles_official