Add train class