added logic for the widget--just needs styling