Cleanup styles and optimize for small screens