Minor updates for styling