Add test for custom templates