Update code element styles