Add class Train