Add Train method