improved the Widget