Add updateStyles method