Redesigned widget.
Fixed some bugs