Improve account UI and form validation