Use flex for form elements