changed training data function input

changed training data function input to have more than 1 training data
point