Add option for specifying template