generate training set
complete exercise 1