Custom component with the sx prop