Fix styling bugs