widgetBuilder -> added buildWidget function