Add complete_style callback