Add Train class