Adding code for custome template