removing sqrt from trainer