Added class Train