adding paralllism to training