Add train class.