Training the data