added train toggle