Add functionality to allow templates