a train method was added