Adding in distinction between test and training rounds