Add support for withField, withButton, etc. helpers