added the possibility for more kinds of training