Add release method to Train class