Fix styling for demo