Add ability to add custom class to main template element