adding tests for custom template
updating tests for custom template