Add support for complete_style callback.