tests added and, training-test sets implemenation fixed