ENH: Add template arg