Updating solver to output training accuracy