Add support for different templates