Adding widget feature