Improved focused element styling