add in template class