Add train method