Add styling for display button