Clean up styles and DOM