More styling tweaks to get everything working