Corrected argument mismatch in add_style_class method