Adding in support for custom classes on the form element div