Clean up styles for demo