Add Train class.