adding remove trains method