More CSS updates to fix styling