Add styling for form states