Fix styling issues I missed