convert to a general train script
Accept a number of test sets.
Adding bias