added the initial training dataset