Add support for custom template file