Fix styling up