Added more logging.  Now records loss and test scores.