Added updateStyle method