Fix styling & mobile UI