Add an optional callback function to override default styling.