Add template logic