Add in styling tweaks