Add template handling