Add template class for tests