improving template with button