update tests and undertrained method