Add tests for custom template