Add option for setting custom template