add training versus testing