Update CSS to fix edge cases