Train GUI updated for completeness