Add include to template to refactor