added training set builders
updated model