Adding test type to the training module