Update code to account for new css