Update for styling changes