added benchmarking for training