Updating mockup to use widget