Fix styling slightly